/*******************************************************************************
 * Copyright 2011-2012 Dik Grapendaal
 * 
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *   http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 ******************************************************************************/
package sh.grapendaal.tsuushin.service.impl;

import java.net.InetSocketAddress;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.Executors;

import org.jboss.netty.bootstrap.ClientBootstrap;
import org.jboss.netty.channel.ChannelFactory;
import org.jboss.netty.channel.socket.nio.NioClientSocketChannelFactory;
import org.jboss.netty.handler.codec.frame.DelimiterBasedFrameDecoder;
import org.jboss.netty.handler.codec.frame.Delimiters;
import org.jboss.netty.handler.codec.string.StringDecoder;
import org.jboss.netty.handler.codec.string.StringEncoder;
import org.jboss.netty.handler.timeout.IdleStateHandler;
import org.jboss.netty.util.CharsetUtil;
import org.jboss.netty.util.HashedWheelTimer;

import sh.grapendaal.tsuushin.core.Network;
import sh.grapendaal.tsuushin.service.NetworkService;

import com.google.inject.Singleton;

@Singleton
public class NetworkServiceImpl implements NetworkService {
	private static final ChannelFactory SOCKET_FACTORY = new NioClientSocketChannelFactory(
			Executors.newCachedThreadPool(), Executors.newCachedThreadPool());
	
	private static final int DEFAULT_PORT = 6667;
	private static final int FRAME_LENGTH = 8192;
	private static final int READER_IDLE_SECONDS = 180;
	private static final int WRITER_IDLE_SECONDS = 0;
	private static final int ALL_IDLE_SECONDS = 0;
	
	private HashedWheelTimer timer;
	private IdleStateHandler idleStateHandler;

	private Map<String, Network> networks = new HashMap<String, Network>();
	
	@Override
	public void start() {
		if (timer == null) {
			timer = new HashedWheelTimer();
			idleStateHandler = new IdleStateHandler(timer, READER_IDLE_SECONDS, WRITER_IDLE_SECONDS, ALL_IDLE_SECONDS);
		}
	}
	
	@Override
	public void stop() {
		SOCKET_FACTORY.releaseExternalResources();
		
		if (timer != null) {
			timer.stop();
			timer = null;
		}
	}
	
	/**
	 * {@inheritDoc}
	 */
	@Override
	public void addServer(String networkName, String serverAddress) {
		this.addServer(networkName, serverAddress, DEFAULT_PORT);
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public void addServer(String networkName, String serverAddress, int port) {
		if (!networks.containsKey(networkName)) {
			networks.put(networkName, new Network(networkName));
		}
		networks.get(networkName).addServer(new InetSocketAddress(serverAddress, port));
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public Network getNetwork(String networkName) {
		return networks.get(networkName);
	}
	
	@Override
	public ClientBootstrap createBootstrap() {
		ClientBootstrap bootstrap = new ClientBootstrap(SOCKET_FACTORY);
		bootstrap.getPipeline().addLast("idle", idleStateHandler);
		bootstrap.getPipeline().addLast("framer", new DelimiterBasedFrameDecoder(FRAME_LENGTH, Delimiters.lineDelimiter()));
		bootstrap.getPipeline().addLast("decoder", new StringDecoder(CharsetUtil.UTF_8));
		bootstrap.getPipeline().addLast("encoder", new StringEncoder(CharsetUtil.UTF_8));
		
		return bootstrap;
	}
}
